package tree.BinarySearchTree;

import tree.TreeNode;

/**
 * @ClassName TrimBst
 * @Description TODO
 * @Author lenovo
 * @Date 2023-04-14 9:52
 * @Version 1.0
 * @Comment Magic. Do not touch.
 * If this comment is removed. the program will blow up
 */
public class TrimBst {
    /**
     * 669. 修剪二叉搜索树
     给你二叉搜索树的根节点 root ，同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树，使得所有节点的值在[low, high]中。修剪树 不应该 改变保留在树中的元素的相对结构 (即，如果没有被移除，原有的父代子代关系都应当保留)。 可以证明，存在 唯一的答案 。

     所以结果应当返回修剪好的二叉搜索树的新的根节点。注意，根节点可能会根据给定的边界发生改变。



     示例 1：


     输入：root = [1,0,2], low = 1, high = 2
     输出：[1,null,2]
     示例 2：
         1
      0    2
     输入：root = [3,0,4,null,2,null,null,1], low = 1, high = 3
     输出：[3,2,null,1]

            3
     0          4
         2
      1
     */
    public TreeNode trimBST(TreeNode root, int low, int high) {
        if(root == null) return null;
        int val = root.val;
        if(root.val<low){
            return trimBST(root.right,low,high);
        }
        if(root.val>high){
            return trimBST(root.left,low,high);
        }
        root.left = trimBST(root.left,low,high);
        root.right= trimBST(root.right,low,high);
        return root;
    }


    public static void main(String[] args) {
        TrimBst trimBst = new TrimBst();
        TreeNode treeNode = new TreeNode(3);
        treeNode.left = new TreeNode(0);
        treeNode.right = new TreeNode(4);
        treeNode.left.right = new TreeNode(2);
        treeNode.left.right.left = new TreeNode(1);
        System.out.println(trimBst.trimBST(treeNode, 1, 3));
    }
}